import random
import time

from calibration_nodes.LoadIon import *
from calibration_nodes.Detection import *
from calibration_nodes.Doppler import *
from calibration_nodes.Pumping import *
from calibration_nodes.EIT import *
from calibration_nodes.ModeFrequency import *
from calibration_nodes.SideBandCooling import *
###此处用户可定义更多结点的py文件###


graph_nodes = [LoadIon, Detection, Doppler, Pumping,  EIT, ModeFrequency, SideBandCooling]

graph_edges = (
    (LoadIon, Detection),
    (Detection, EIT),
    (LoadIon, Doppler),
    (Doppler, EIT),
    (LoadIon, Pumping),
    (Pumping, EIT),
    (EIT, ModeFrequency),
    (ModeFrequency, SideBandCooling)
)

core_nodes = [Detection, SideBandCooling]
